#!/bin/bash

# Docker方式启动视频分析系统

echo "🐳 使用Docker启动视频分析系统"
echo "================================"

# 检查Docker
if ! command -v docker &> /dev/null; then
    echo "❌ Docker 未安装，请先安装Docker"
    exit 1
fi

# 检查Docker Compose
if ! command -v docker-compose &> /dev/null; then
    echo "❌ Docker Compose 未安装，请先安装Docker Compose"
    exit 1
fi

# 检查NVIDIA Docker支持
if ! docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi &> /dev/null; then
    echo "⚠️  警告: NVIDIA Docker支持未配置，GPU加速可能不可用"
    echo "   请安装 nvidia-container-toolkit"
else
    echo "✅ NVIDIA Docker支持已配置"
fi

# 创建必要的目录
echo "📁 创建必要的目录..."
mkdir -p uploads frames

# 拉取最新镜像
echo "📥 拉取Docker镜像..."
docker-compose pull

# 构建应用镜像
echo "🔨 构建应用镜像..."
docker-compose build

# 启动服务
echo "🚀 启动所有服务..."
docker-compose up -d

# 等待服务启动
echo "⏳ 等待服务启动..."
sleep 10

# 检查服务状态
echo "🔍 检查服务状态..."
docker-compose ps

# 显示访问信息
echo ""
echo "✅ 服务启动完成!"
echo "🌐 Web应用: http://localhost"
echo "🤖 模型API: http://localhost:8001"
echo "📊 应用API: http://localhost:8000"
echo ""
echo "📝 查看日志:"
echo "   docker-compose logs -f app"
echo "   docker-compose logs -f model-server"
echo ""
echo "🛑 停止服务:"
echo "   docker-compose down"
